from .gan import Generator